# -*- coding: utf-8 -*-
# @Author  : longbhu
# @Time    : 2025/3/26 11:55
# @Function:
import os

def remove_field_from_filenames(folder_path, file_extension, field_to_remove):
    """
    批量去除指定文件夹中指定扩展名文件名中的指定字段

    :param folder_path: 文件夹路径
    :param file_extension: 文件扩展名，例如 '.tif'
    :param field_to_remove: 要去除的字段，例如 '_excel'
    """
    # 获取所有文件名
    file_names = os.listdir(folder_path)

    # 新文件名列表
    new_file_names = []

    # 遍历所有文件名
    for file_name in file_names:
        # 检查文件是否为指定扩展名
        if file_name.endswith(file_extension):
            # 检查文件名中是否包含要去除的字段
            if field_to_remove in file_name:
                # 去除字段
                new_file_name = file_name.replace(field_to_remove, '')
            else:
                # 如果不包含，保持原样
                new_file_name = file_name
        else:
            # 如果不是指定扩展名的文件，保持原样
            new_file_name = file_name

        # 重命名文件
        os.rename(os.path.join(folder_path, file_name), os.path.join(folder_path, new_file_name))
        new_file_names.append(new_file_name)

    # 输出新的文件名列表
    return new_file_names

# 文件夹路径
folder_path = r'H:/GEP_result/2023/Qwp'

# 调用函数，去除.tif文件名中的 '_excel'
new_file_names = remove_field_from_filenames(folder_path, '.tif', '_excel')

# 输出新的文件名列表
print(new_file_names)